package cn.birdbaby.plugins.refactorgen;

import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiPackage;
import com.intellij.refactoring.listeners.RefactoringElementListener;
import com.intellij.refactoring.listeners.RefactoringElementListenerProvider;

public class BBRefactoringElementListenerProvider implements RefactoringElementListenerProvider {
    @Override
    public RefactoringElementListener getListener(PsiElement element) {
        if (element instanceof PsiClass||element instanceof PsiPackage) {
            return new BBRefactoringElementListener();
        }
        return null;
    }
}